package com.jxb.dp20251355;

/**
 * 类注释
 *
 * @author jiaobo
 * @date Created in 2025/1/7 21:06
 **/
public class BuyStockWithCooldown1_309 {

    //dp[i][j]表示某天的收益，i表示第几天，j表示股票状态
    //j=0,卖出不持有
    //j=1，未动作，不持有
    //j=2，买入，持有
    //j=3，冷冻期，不持有
    //状态转移公式如下：
    //  dp[i][0] = prices[i] + dp[i-1][2]
    //  dp[i][1] = max(dp[i-1][1],dp[i-1][3])
    //  dp[i][2] = max(dp[i-1]-prices[i],dp[i-1][2],dp[i-1][3]-prices[i])
    //  dp[i][3] = dp[i-1][0]
    public int maxProfit(int[] prices) {
        int length = prices.length;
        //定义dp数组
        int[][] dp = new int[length][4];
        dp[0][2] = -prices[0];

        for (int i = 1;i<length;i++) {
            dp[i][0] = prices[i] + dp[i-1][2];
            dp[i][1] = Math.max(dp[i-1][1],dp[i-1][3]);
            dp[i][2] = Math.max(dp[i-1][1]-prices[i],Math.max(dp[i-1][2],dp[i-1][3]-prices[i]));
            dp[i][3] = dp[i-1][0];
        }
        return Math.max(dp[length-1][0],Math.max(dp[length-1][1],dp[length-1][3]));
    }

    /*DP+滚动数组实现*/
    public int maxProfit2(int[] prices) {
        int max = prices[0];

        for (int i = 1;i<prices.length;i++) {
            //前一天买入，第二天卖出：prices[i] - prices[i-1]
            //前一天未动作，不持有，第二天不持有：0
            //前一天未动作，不持有，第二天买入：-prices[i]
            //前一天未动作,冷冻期，不持有，第二天不持有：-prices[i-1]
            //前一天未动作,冷冻期，不持有，第二天持有：-prices[i-1]-prices[i]
            int curMax = Math.max(prices[i] - prices[i-1],Math.max(0,Math.max(-prices[i],Math.max(-prices[i-1],
                                                                                     -prices[i-1]-prices[i]))));
            max = Math.max(curMax,max);
        }
        return max;
    }

}
